"""
    将 model 注册到 xadmin
"""
import xadmin
from .models import UserAsk, UserCourse, UserMessage, CourseComments, UserFavorite


# 用户咨询
# 这里必须写一个基础 object
class UserAskAdmin(object):
    # 控制 xadmin 中显示哪些字段
    list_display = ['name', 'mobile', 'course_name', 'add_time']
    # 设置 xadmin 中可搜索的字段
    search_fields = ['name', 'mobile', 'course_name']
    # 设置可筛选的字段
    list_filter = ['name', 'mobile', 'course_name', 'add_time']


# 用户学习的课程
# 这里必须写一个基础 object
class UserCourseAdmin(object):
    # 控制 xadmin 中显示哪些字段
    list_display = ['user', 'course', 'add_time']
    # 设置 xadmin 中可搜索的字段
    search_fields = ['user', 'course']
    # 设置可筛选的字段
    list_filter = ['user', 'course', 'add_time']


# 课程评论
# 这里必须写一个基础 object
class CourseCommentsAdmin(object):
    # 控制 xadmin 中显示哪些字段
    list_display = ['user', 'course', 'comments', 'add_time']
    # 设置 xadmin 中可搜索的字段
    search_fields = ['user', 'course', 'comments']
    # 设置可筛选的字段
    list_filter = ['user', 'course', 'comments', 'add_time']


# 用户消息
# 这里必须写一个基础 object
class UserMessageAdmin(object):
    # 控制 xadmin 中显示哪些字段
    list_display = ['user', 'message', 'has_read', 'add_time']
    # 设置 xadmin 中可搜索的字段
    search_fields = ['user', 'message', 'has_read']
    # 设置可筛选的字段
    list_filter = ['user', 'message', 'has_read', 'add_time']


# 用户收藏
# 这里必须写一个基础 object
class UserFavoriteAdmin(object):
    # 控制 xadmin 中显示哪些字段
    list_display = ['user', 'fav_id', 'fav_type', 'add_time']
    # 设置 xadmin 中可搜索的字段
    search_fields = ['user', 'fav_id', 'fav_type']
    # 设置可筛选的字段
    list_filter = ['user', 'fav_id', 'fav_type', 'add_time']


# 将 model 注册到 xadmin
xadmin.site.register(UserAsk, UserAskAdmin)
xadmin.site.register(UserCourse, UserCourseAdmin)
xadmin.site.register(CourseComments, CourseCommentsAdmin)
xadmin.site.register(UserMessage, UserMessageAdmin)
xadmin.site.register(UserFavorite, UserFavoriteAdmin)